import pathlib

import spinn_trainer
from checkpoint_manager import CheckpointManager
from simple_pendulum_dataset import SimplePendulumDataset


def main():
    title = "not_damped"
    spinn_trainer.train(
        CheckpointManager(pathlib.Path(f"./checkpoints/{title}")),
        SimplePendulumDataset.load(pathlib.Path(f"./data/{title}.pt"))
    )


if __name__ == "__main__":
    main()
